import os
from pydub import AudioSegment

def split_audio(input_file, output_dir, start_time, end_time):
    
    # 打开原始音频文件
    sound = AudioSegment.from_wav("data/audio/"+input_file+".wav")

   
    # 截取音频
    extracted_sound = sound[start_time:end_time]

    # 导出为新的文件
    extracted_sound.export(output_dir+input_file+'-50s.wav', format="wav")

    print("提取完成！")

                
if __name__ == '__main__':
    for i in range(1,11):
        if i < 10:
            input_file = 'run166-A1P0' + str(i)
        else:
            input_file = 'run166-A1P' + str(i)
        output_dir = 'data/audio_train/run166/'
        # 定义截取的起止时间
        # start_time = 0 * 60 * 1000 + 25 * 1000  # 起始时间：2分5秒
        # end_time = 1 * 60 * 1000 + 15 * 1000   # 终止时间：2分55秒
        
        # start_time = 0 * 60 * 1000 + 25 * 1000  # 起始时间：2分5秒
        # end_time = 1 * 60 * 1000 + 15 * 1000   # 终止时间：2分55秒
        
        # start_time = 0 * 60 * 1000 + 25 * 1000  # 起始时间：2分5秒
        # end_time = 1 * 60 * 1000 + 15 * 1000   # 终止时间：2分55秒
        
        start_time = 1 * 60 * 1000 + 5 * 1000  # 起始时间：2分5秒
        end_time = 1 * 60 * 1000 + 55 * 1000   # 终止时间：2分55秒
        split_audio(input_file, output_dir, start_time, end_time)
    